CV

sealed class CV : OnnxModelType<ImageRecognitionModel>

Image recognition models and preprocessing.

Types

EfficientNet4Lite

jvm

object EfficientNet4Lite : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB0

jvm

class EfficientNetB0(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB1

jvm

class EfficientNetB1(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB2

jvm

class EfficientNetB2(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB3

jvm

class EfficientNetB3(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB4

jvm

class EfficientNetB4(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB5

jvm

class EfficientNetB5(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB6

jvm

class EfficientNetB6(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

EfficientNetB7

jvm

class EfficientNetB7(noTop: Boolean) : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

Lenet

jvm

class Lenet : ONNXModels.CV

This model is a neural network for digit classification that take grey-scale images of digits as input and classify the major object in the image into a set of 10 different classes.

ResNet101

jvm

object ResNet101 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet101v2

jvm

object ResNet101v2 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet152

jvm

object ResNet152 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet152v2

jvm

object ResNet152v2 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet18

jvm

object ResNet18 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet18v2

jvm

object ResNet18v2 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet34

jvm

object ResNet34 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet34v2

jvm

object ResNet34v2 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet50

jvm

object ResNet50 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet50custom

jvm

object ResNet50custom : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet50noTopCustom

jvm

object ResNet50noTopCustom : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

ResNet50v2

jvm

object ResNet50v2 : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

Functions

model

jvm

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

jvm

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Properties

inputShape

jvm

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

jvm

open override val modelRelativePath: String

preprocessor

jvm

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV